
import glob
import csv
import sys, os.path

import sklearn
import torch
from torch import optim, nn
from torch.utils.data import DataLoader
from torchvision import transforms
#from torch.utils.tensorboard import SummaryWriter
#from torchsummary import summary
import numpy as np
import matplotlib.pyplot as plt
from tqdm import tqdm
import warnings
warnings.filterwarnings('ignore')


# Our modules
import sys
sys.path.append('.')
sys.path.append('..')

from vae import configs, train, plot_utils, models
from vae.data import build_dataloader
from vae.latent_spaces import dimensionality_reduction, plot_spaces
from vae.reconstructions import plot_reconstructions
from vae.models import model_utils
from vae.train import training


torch.__version__

'0.4.1.post2'


!lspci | grep -i nvidia

01:00.0 VGA compatible controller: NVIDIA Corporation GM200 [GeForce GTX 980 Ti] (rev a1)
01:00.1 Audio device: NVIDIA Corporation GM200 High Definition Audio (rev a1)


torch.cuda.is_available()

True


torch.backends.cudnn.enabled

True


!nvidia-smi

Wed Apr 14 11:52:25 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.56       Driver Version: 460.56       CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce GTX 980 Ti  Off  | 00000000:01:00.0 Off |                  N/A |
| 45%   68C    P0    92W / 250W |     48MiB /  6080MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      6989      G   /usr/bin/X                         27MiB |
|    0   N/A  N/A      7679      G   /usr/bin/gnome-shell               14MiB |
+-----------------------------------------------------------------------------+


tensorboard = False
if tensorboard:
    # Load the TensorBoard notebook extension
    %load_ext tensorboard
    %tensorboard --logdir ../runs


training.train_model(input='mel_cut', model_name='TimeConv2D_cut', loss_f='bce')

Building dataloaders...
Dataloaders have been built.
Number of files in the training dataset: 11630
Number of files in the validation dataset: 1357

  0%|          | 0/727 [00:00<?, ?it/s]

Start training model on cuda ...
Epoch: 0

100%|██████████| 727/727 [00:09<00:00, 79.46it/s]

training_avg_loss=16729.18
 training_avg_kld1=373663.57
 training_avg_kld2=373663.57
 training_avg_rec=15930.31

val_avg_loss=5703.15

  0%|          | 0/727 [00:00<?, ?it/s]

Epoch: 1

100%|██████████| 727/727 [00:09<00:00, 76.34it/s]

CNN VAE ¶

Table of Contents ¶

0. Check GPU and Troch versions ¶

4. RUN TRAINING ¶

CNN VAE ¶

Table of Contents ¶

0. Check GPU and Troch versions¶

4. RUN TRAINING ¶

0. Check GPU and Troch versions ¶